#pragma once

#include <brpc/server.h>
#include <butil/logging.h>
#include <regex>
#include <jsoncpp/json/json.h>

#include "data_es.h"    // es数据管理客户端封装
#include "data_redis.h" // redis数据管理客户端封装
#include "mysql_user.h" // mysql数据管理客户端封装
#include "etcd.h"       // 服务注册模块封装
#include "logger.h"     // 日志模块封装
#include "utils.h"      // 基础工具接口
#include "dms.h"        // 短信平台SDK模块封装
#include "channel.h"    // 信道管理模块封装

#include "user.hxx"
#include "user-odb.hxx" // odb框架代码

#include "user.pb.h" // protobuf框架代码
#include "base.pb.h" // protobuf框架代码
#include "file.pb.h" // protobuf框架代码

class UserServiceImpl : public ljm::UserService
{
public:
    UserServiceImpl(const DMSClient::ptr &dms_client,
                    const std::shared_ptr<elasticlient::Client> &es_client,
                    const std::shared_ptr<odb::core::database> &mysql_client,
                    const std::shared_ptr<sw::redis::Redis> &redis_client,
                    const ServiceManager::ptr &channel_manager,
                    const std::string &file_service_name) : _es_user(std::make_shared<ESUser>(es_client)),
                                                            _mysql_user(std::make_shared<UserTable>(mysql_client)),
                                                            _redis_session(std::make_shared<Session>(redis_client)),
                                                            _redis_status(std::make_shared<Status>(redis_client)),
                                                            _redis_codes(std::make_shared<Codes>(redis_client)),
                                                            _file_service_name(file_service_name),
                                                            _sm_channels(channel_manager),
                                                            _dms_client(dms_client)
    {
        // 创建用户元信息索引
        _es_user->createIndex();
    }

    // 检查用户昵称是否合法
    bool nickname_check(const std::string &nickname)
    {
        return nickname.size() < 22;
    }

    // 检查用户密码是否合法
    bool password_check(const std::string &password)
    {
        // 正则表达式：匹配长度为8-20，并且包含大小写字母、数字、特殊字符
        std::regex passwordRegex(
            R"(^(?=.*[A-Z])(?=.*[a-z])(?=.*\d)(?=.*[!@#$%^&*])[A-Za-z\d!@#$%^&*]{8,20}$)");

        // 检查密码是否匹配正则表达式
        return std::regex_match(password, passwordRegex);
    }

    // 用户用户名密码注册
    void UserRegister(::google::protobuf::RpcController *controller,
                      const ::ljm::UserRegisterReq *request,
                      ::ljm::UserRegisterRsp *response,
                      ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到用户注册请求！");
        brpc::ClosureGuard rpc_guard(done);
        // 定义一个错误处理函数，当出错的时候被调用
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };

        // 1. 从请求中取出昵称和密码
        std::string nickname = request->nickname();
        std::string password = request->password();
        // 2. 检查昵称是否合法（长度小于22字节）
        bool ret = nickname_check(nickname);
        if (!ret)
        {
            LOG_ERROR("{} - 用户名长度不合法！", request->request_id());
            return err_response(request->request_id(), "用户名长度不合法！");
        }
        // 3. 检查密码是否合法（匹配长度为8-20，并且包含大小写字母、数字、特殊字符）
        ret = password_check(password);
        if (!ret)
        {
            LOG_ERROR("{} - 密码格式不合法！", request->request_id());
            return err_response(request->request_id(), "密码格式不合法！");
        }
        // 4. 根据昵称在数据库进行判断是否昵称已存在
        auto user = _mysql_user->select_by_nickname(nickname);
        if (user)
        {
            LOG_ERROR("{} - 用户名被占用- {}！", request->request_id(), nickname);
            return err_response(request->request_id(), "用户名被占用!");
        }
        // 5. 向数据库新增数据
        std::string uid = uuid();
        user = std::make_shared<User>(uid, nickname, password);
        ret = _mysql_user->insert(user);
        if (!ret)
        {
            LOG_ERROR("{} - Mysql数据库新增数据失败！", request->request_id());
            return err_response(request->request_id(), "Mysql数据库新增数据失败!");
        }
        // 6. 向 ES 服务器中新增用户信息
        ret = _es_user->appendData(uid, "", nickname, "", "");
        if (!ret)
        {
            LOG_ERROR("{} - ES搜索引擎新增数据失败！", request->request_id());
            return err_response(request->request_id(), "ES搜索引擎新增数据失败！");
        }
        // 7. 组织响应，进行成功与否的响应即可。
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 用户登录
    void UserLogin(::google::protobuf::RpcController *controller,
                   const ::ljm::UserLoginReq *request,
                   ::ljm::UserLoginRsp *response,
                   ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到用户登录请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出昵称和密码
        std::string nickname = request->nickname();
        std::string password = request->password();
        // 2. 通过昵称获取用户信息，进行密码是否一致的判断
        auto user = _mysql_user->select_by_nickname(nickname);
        if (!user || password != user->password())
        {
            LOG_ERROR("{} - 用户名或密码错误 - {}-{}！", request->request_id(), nickname, password);
            return err_response(request->request_id(), "用户名或密码错误!");
        }
        // 3. 根据 redis 中的登录标记信息是否存在判断用户是否已经登录。
        bool ret = _redis_status->exists(user->user_id());
        if (ret)
        {
            LOG_ERROR("{} - 用户已在其他地方登录 - {}！", request->request_id(), nickname);
            return err_response(request->request_id(), "用户已在其他地方登录!");
        }
        // 4. 构造会话 ID，生成会话键值对，向 redis 中添加会话信息以及登录标记信息
        std::string ssid = uuid();
        _redis_session->append(ssid, user->user_id());
        // 5. 添加用户登录信息
        _redis_status->append(user->user_id());
        // 5. 组织响应，返回生成的会话 ID
        response->set_request_id(request->request_id());
        response->set_login_session_id(ssid);
        response->set_success(true);
    }

    // 检查手机号是否正确
    bool phone_check(const std::string &phone)
    {
        if (phone.size() != 11)
            return false;
        if (phone[0] != '1')
            return false;
        if (phone[1] < '3' || phone[1] > '9')
            return false;
        for (int i = 2; i < 11; i++)
        {
            if (phone[i] < '0' || phone[i] > '9')
                return false;
        }
        return true;
    }

    // 获取验证码
    void GetPhoneVerifyCode(::google::protobuf::RpcController *controller,
                            const ::ljm::PhoneVerifyCodeReq *request,
                            ::ljm::PhoneVerifyCodeRsp *response,
                            ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到短信验证码获取请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出手机号码
        std::string phone = request->phone_number();
        // 2. 验证手机号码格式是否正确（必须以 1 开始，第二位 3~9 之间，后边 9 个数字字符）
        bool ret = phone_check(phone);
        if (!ret)
        {
            LOG_ERROR("{} - 手机号码格式错误 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "手机号码格式错误!");
        }
        // 3. 生成 4 位随机验证码
        std::string code_id = uuid();
        std::string code = getCode();
        // 4. 基于短信平台 SDK 发送验证码
        ret = _dms_client->send(phone, code);
        if (!ret)
        {
            LOG_ERROR("{} - 短信验证码发送失败 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "短信验证码发送失败!");
        }
        // 5. 构造验证码 ID，添加到 redis 验证码映射键值索引中
        _redis_codes->append(code_id, code);
        // 6. 组织响应，返回生成的验证码 ID
        response->set_request_id(request->request_id());
        response->set_success(true);
        response->set_verify_code_id(code_id);
        LOG_DEBUG("获取短信验证码处理完成！");
    }

    // 使用手机注册
    void PhoneRegister(::google::protobuf::RpcController *controller,
                       const ::ljm::PhoneRegisterReq *request,
                       ::ljm::PhoneRegisterRsp *response,
                       ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到手机号注册请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出手机号码和验证码,验证码ID
        std::string phone = request->phone_number();
        std::string code_id = request->verify_code_id();
        std::string code = request->verify_code();
        // 2. 检查注册手机号码是否合法
        bool ret = phone_check(phone);
        if (!ret)
        {
            LOG_ERROR("{} - 手机号码格式错误 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "手机号码格式错误!");
        }
        // 3. 从 redis 数据库中进行验证码 ID-验证码一致性匹配
        auto vcode = _redis_codes->code(code_id);
        if (vcode != code)
        {
            LOG_ERROR("{} - 验证码错误 - {}-{}！", request->request_id(), code_id, code);
            return err_response(request->request_id(), "验证码错误!");
        }
        // 4. 通过数据库查询判断手机号是否已经注册过
        auto user = _mysql_user->select_by_phone(phone);
        if (user)
        {
            LOG_ERROR("{} - 该手机号已注册过用户 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "该手机号已注册过用户!");
        }
        // 5. 向数据库新增用户信息
        std::string uid = uuid();
        user = std::make_shared<User>(uid, phone);
        ret = _mysql_user->insert(user);
        if (!ret)
        {
            LOG_ERROR("{} - 向数据库添加用户信息失败 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "向数据库添加用户信息失败!");
        }
        // 6. 向 ES 服务器中新增用户信息
        ret = _es_user->appendData(uid, phone, uid, "", "");
        if (ret == false)
        {
            LOG_ERROR("{} - ES搜索引擎新增数据失败！", request->request_id());
            return err_response(request->request_id(), "ES搜索引擎新增数据失败！");
        }
        // 7. 组织响应，进行成功与否的响应即可。
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 使用手机登录
    void PhoneLogin(::google::protobuf::RpcController *controller,
                    const ::ljm::PhoneLoginReq *request,
                    ::ljm::PhoneLoginRsp *response,
                    ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到手机号登录请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出手机号码和验证码 ID，以及验证码。
        std::string phone = request->phone_number();
        std::string code_id = request->verify_code_id();
        std::string code = request->verify_code();
        // 2. 检查注册手机号码是否合法
        bool ret = phone_check(phone);
        if (ret == false)
        {
            LOG_ERROR("{} - 手机号码格式错误 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "手机号码格式错误!");
        }
        // 3. 根据手机号从数据数据进行用户信息查询，判断用用户是否存在
        auto user = _mysql_user->select_by_phone(phone);
        if (!user)
        {
            LOG_ERROR("{} - 该手机号未注册用户 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "该手机号未注册用户!");
        }
        // 4. 从 redis 数据库中进行验证码 ID-验证码一致性匹配
        auto vcode = _redis_codes->code(code_id);
        if (vcode != code)
        {
            LOG_ERROR("{} - 验证码错误 - {}-{}！", request->request_id(), code_id, code);
            return err_response(request->request_id(), "验证码错误!");
        }
        _redis_codes->remove(code_id);
        // 5. 根据 redis 中的登录标记信息是否存在判断用户是否已经登录。
        ret = _redis_status->exists(user->user_id());
        if (ret == true)
        {
            LOG_ERROR("{} - 用户已在其他地方登录 - {}！", request->request_id(), phone);
            return err_response(request->request_id(), "用户已在其他地方登录!");
        }
        // 4. 构造会话 ID，生成会话键值对，向 redis 中添加会话信息以及登录标记信息
        std::string ssid = uuid();
        _redis_session->append(ssid, user->user_id());
        // 5. 添加用户登录信息
        _redis_status->append(user->user_id());
        // 7. 组织响应，返回生成的会话 ID
        response->set_request_id(request->request_id());
        response->set_login_session_id(ssid);
        response->set_success(true);
    }

    // 获取用户信息
    void GetUserInfo(::google::protobuf::RpcController *controller,
                     const ::ljm::GetUserInfoReq *request,
                     ::ljm::GetUserInfoRsp *response,
                     ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到获取单个用户信息请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出用户 ID
        std::string uid = request->user_id();
        // 2. 通过用户 ID，从数据库中查询用户信息
        auto user = _mysql_user->select_by_id(uid);
        if (!user)
        {
            LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
            return err_response(request->request_id(), "未找到用户信息!");
        }
        // 3. 根据用户信息中的头像 ID，从文件服务器获取头像文件数据，组织完整用户信息
        ljm::UserInfo *user_info = response->mutable_user_info();
        user_info->set_user_id(user->user_id());
        user_info->set_nickname(user->nickname());
        user_info->set_description(user->description());
        user_info->set_phone(user->phone());
        if (!user->avatar_id().empty())
        {
            // 从信道管理对象中，获取到连接了文件管理子服务的channel
            auto channel = _sm_channels->getService(_file_service_name);
            if (!channel)
            {
                LOG_ERROR("{} - 未找到文件管理子服务节点 - {} - {}！",
                          request->request_id(), _file_service_name, uid);
                return err_response(request->request_id(), "未找到文件管理子服务节点!");
            }
            // 进行文件子服务的rpc请求，进行头像文件下载
            ljm::FileService_Stub stub(channel.get());
            ljm::GetSingleFileReq req;
            ljm::GetSingleFileRsp rsp;
            req.set_request_id(request->request_id());
            req.set_file_id(user->avatar_id());
            brpc::Controller cntl;
            stub.GetSingleFile(&cntl, &req, &rsp, nullptr);
            if (cntl.Failed() || !rsp.success())
            {
                LOG_ERROR("{} - 文件子服务调用失败：{}！", request->request_id(), cntl.ErrorText());
                return err_response(request->request_id(), "文件子服务调用失败!");
            }
            user_info->set_avatar(rsp.file_data().file_content());
        }
        // 4. 组织响应，返回用户信息
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 获取多组用户信息
    void GetMultiUserInfo(::google::protobuf::RpcController *controller,
                          const ::ljm::GetMultiUserInfoReq *request,
                          ::ljm::GetMultiUserInfoRsp *response,
                          ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到批量用户信息获取请求！");
        brpc::ClosureGuard rpc_guard(done);
        // 1. 定义错误回调
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 2. 从请求中取出用户ID --- 列表
        std::vector<std::string> uid_lists;
        for (int i = 0; i < request->users_id_size(); i++)
        {
            uid_lists.push_back(request->users_id(i));
        }
        // 3. 从数据库进行批量用户信息查询
        auto users = _mysql_user->select_multi_users(uid_lists);
        if (users.size() != request->users_id_size())
        {
            LOG_ERROR("{} - 从数据库查找的用户信息数量不一致 {}-{}！",
                      request->request_id(), request->users_id_size(), users.size());
            return err_response(request->request_id(), "从数据库查找的用户信息数量不一致!");
        }
        // 4. 批量从文件管理子服务进行文件下载
        auto channel = _sm_channels->getService(_file_service_name);
        if (!channel)
        {
            LOG_ERROR("{} - 未找到文件管理子服务节点 - {}！", request->request_id(), _file_service_name);
            return err_response(request->request_id(), "未找到文件管理子服务节点!");
        }
        ljm::FileService_Stub stub(channel.get());
        ljm::GetMultiFileReq req;
        ljm::GetMultiFileRsp rsp;
        req.set_request_id(request->request_id());
        for (auto &user : users)
        {
            if (user.avatar_id().empty())
                continue;
            req.add_file_id_list(user.avatar_id());
        }
        brpc::Controller cntl;
        stub.GetMultiFile(&cntl, &req, &rsp, nullptr);
        if (cntl.Failed() || !rsp.success())
        {
            LOG_ERROR("{} - 文件子服务调用失败：{} - {}！", request->request_id(),
                      _file_service_name, cntl.ErrorText());
            return err_response(request->request_id(), "文件子服务调用失败!");
        }
        // 5. 组织响应（）
        for (auto &user : users)
        {
            auto user_map = response->mutable_users_info(); // 本次请求要响应的用户信息map
            auto file_map = rsp.mutable_file_data();        // 这是批量文件请求响应中的map
            ljm::UserInfo user_info;
            user_info.set_user_id(user.user_id());
            user_info.set_nickname(user.nickname());
            user_info.set_description(user.description());
            user_info.set_phone(user.phone());
            user_info.set_avatar((*file_map)[user.avatar_id()].file_content());
            (*user_map)[user_info.user_id()] = user_info;
        }
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 修改头像数据
    void SetUserAvatar(::google::protobuf::RpcController *controller,
                       const ::ljm::SetUserAvatarReq *request,
                       ::ljm::SetUserAvatarRsp *response,
                       ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到用户头像设置请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出用户 ID 与头像数据
        std::string uid = request->user_id();
        // 2. 从数据库通过用户 ID 进行用户信息查询，判断用户是否存在
        auto user = _mysql_user->select_by_id(uid);
        if (!user)
        {
            LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
            return err_response(request->request_id(), "未找到用户信息!");
        }
        // 3. 上传头像文件到文件子服务，
        auto channel = _sm_channels->getService(_file_service_name);
        if (!channel)
        {
            LOG_ERROR("{} - 未找到文件管理子服务节点 - {}！", request->request_id(), _file_service_name);
            return err_response(request->request_id(), "未找到文件管理子服务节点!");
        }
        ljm::FileService_Stub stub(channel.get());
        ljm::PutSingleFileReq req;
        ljm::PutSingleFileRsp rsp;
        req.set_request_id(request->request_id());
        req.mutable_file_data()->set_file_name("");
        req.mutable_file_data()->set_file_size(request->avatar().size());
        req.mutable_file_data()->set_file_content(request->avatar());
        brpc::Controller cntl;
        stub.PutSingleFile(&cntl, &req, &rsp, nullptr);
        if (cntl.Failed() || !rsp.success())
        {
            LOG_ERROR("{} - 文件子服务调用失败：{}！", request->request_id(), cntl.ErrorText());
            return err_response(request->request_id(), "文件子服务调用失败!");
        }
        std::string avatar_id = rsp.file_info().file_id();
        // 4. 将返回的头像文件 ID 更新到数据库中
        user->avatar_id(avatar_id);
        bool ret = _mysql_user->update(user);
        if (!ret)
        {
            LOG_ERROR("{} - 更新数据库用户头像ID失败 ：{}！", request->request_id(), avatar_id);
            return err_response(request->request_id(), "更新数据库用户头像ID失败!");
        }
        // 5. 更新 ES 服务器中用户信息
        ret = _es_user->appendData(user->user_id(), user->phone(),
                                   user->nickname(), user->description(), user->avatar_id());
        if (!ret)
        {
            LOG_ERROR("{} - 更新搜索引擎用户头像ID失败 ：{}！", request->request_id(), avatar_id);
            return err_response(request->request_id(), "更新搜索引擎用户头像ID失败!");
        }
        // 6. 组织响应，返回更新成功与否
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 修改用户昵称
    void SetUserNickname(::google::protobuf::RpcController *controller,
                         const ::ljm::SetUserNicknameReq *request,
                         ::ljm::SetUserNicknameRsp *response,
                         ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到用户昵称设置请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出用户 ID 与新的昵称
        std::string uid = request->user_id();
        std::string new_nickname = request->nickname();
        // 2. 判断昵称格式是否正确
        bool ret = nickname_check(new_nickname);
        if (!ret)
        {
            LOG_ERROR("{} - 用户名长度不合法！", request->request_id());
            return err_response(request->request_id(), "用户名长度不合法！");
        }
        // 3. 从数据库通过用户 ID 进行用户信息查询，判断用户是否存在
        auto user = _mysql_user->select_by_id(uid);
        if (!user)
        {
            LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
            return err_response(request->request_id(), "未找到用户信息!");
        }
        // 4. 将新的昵称更新到数据库中
        user->nickname(new_nickname);
        ret = _mysql_user->update(user);
        if (!ret)
        {
            LOG_ERROR("{} - 更新数据库用户昵称失败 ：{}！", request->request_id(), new_nickname);
            return err_response(request->request_id(), "更新数据库用户昵称失败!");
        }
        // 5. 更新 ES 服务器中用户信息
        ret = _es_user->appendData(user->user_id(), user->phone(),
                                   user->nickname(), user->description(), user->avatar_id());
        if (!ret)
        {
            LOG_ERROR("{} - 更新搜索引擎用户昵称失败 ：{}！", request->request_id(), new_nickname);
            return err_response(request->request_id(), "更新搜索引擎用户昵称失败!");
        }
        // 6. 组织响应，返回更新成功与否
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 修改用户描述
    void SetUserDescription(::google::protobuf::RpcController *controller,
                            const ::ljm::SetUserDescriptionReq *request,
                            ::ljm::SetUserDescriptionRsp *response,
                            ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到用户签名设置请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出用户 ID 与新的用户描述
        std::string uid = request->user_id();
        std::string new_description = request->description();
        // 3. 从数据库通过用户 ID 进行用户信息查询，判断用户是否存在
        auto user = _mysql_user->select_by_id(uid);
        if (!user)
        {
            LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
            return err_response(request->request_id(), "未找到用户信息!");
        }
        // 4. 将新的用户描述更新到数据库中
        user->description(new_description);
        bool ret = _mysql_user->update(user);
        if (!ret)
        {
            LOG_ERROR("{} - 更新数据库用户签名失败 ：{}！", request->request_id(), new_description);
            return err_response(request->request_id(), "更新数据库用户签名失败!");
        }
        // 5. 更新 ES 服务器中用户信息
        ret = _es_user->appendData(user->user_id(), user->phone(),
                                   user->nickname(), user->description(), user->avatar_id());
        if (!ret)
        {
            LOG_ERROR("{} - 更新搜索引擎用户签名失败 ：{}！", request->request_id(), new_description);
            return err_response(request->request_id(), "更新搜索引擎用户签名失败!");
        }
        // 6. 组织响应，返回更新成功与否
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 修改用户手机号
    void SetUserPhoneNumber(::google::protobuf::RpcController *controller,
                            const ::ljm::SetUserPhoneNumberReq *request,
                            ::ljm::SetUserPhoneNumberRsp *response,
                            ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到用户手机号设置请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出用户 ID 与新的手机号
        std::string uid = request->user_id();
        std::string new_phone = request->phone_number();
        std::string code = request->phone_verify_code();
        std::string code_id = request->phone_verify_code_id();
        // 2. 对验证码进行验证
        auto vcode = _redis_codes->code(code_id);
        if (vcode != code)
        {
            LOG_ERROR("{} - 验证码错误 - {}-{}！", request->request_id(), code_id, code);
            return err_response(request->request_id(), "验证码错误!");
        }
        // 3. 从数据库通过用户 ID 进行用户信息查询，判断用户是否存在
        auto user = _mysql_user->select_by_id(uid);
        if (!user)
        {
            LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
            return err_response(request->request_id(), "未找到用户信息!");
        }
        // 4. 将新的手机号更新到数据库中
        user->phone(new_phone);
        bool ret = _mysql_user->update(user);
        if (!ret)
        {
            LOG_ERROR("{} - 更新数据库用户手机号失败 ：{}！", request->request_id(), new_phone);
            return err_response(request->request_id(), "更新数据库用户手机号失败!");
        }
        // 5. 更新 ES 服务器中用户信息
        ret = _es_user->appendData(user->user_id(), user->phone(),
                                   user->nickname(), user->description(), user->avatar_id());
        if (!ret)
        {
            LOG_ERROR("{} - 更新搜索引擎用户手机号失败 ：{}！", request->request_id(), new_phone);
            return err_response(request->request_id(), "更新搜索引擎用户手机号失败!");
        }
        // 6. 组织响应，返回更新成功与否
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    // 修改用户密码
    void SetUserPassword(::google::protobuf::RpcController *controller,
                         const ::ljm::SetUserPasswordReq *request,
                         ::ljm::SetUserPasswordRsp *response,
                         ::google::protobuf::Closure *done)
    {
        LOG_DEBUG("收到用户密码设置请求！");
        brpc::ClosureGuard rpc_guard(done);
        auto err_response = [this, response](const std::string &rid,
                                             const std::string &errmsg) -> void
        {
            response->set_request_id(rid);
            response->set_success(false);
            response->set_errmsg(errmsg);
            return;
        };
        // 1. 从请求中取出用户 ID 与新的密码 ---- 该密码后期会改为密码的md5值, 不存储真实密码
        std::string uid = request->user_id();
        std::string new_pwd = request->password();
        // std::string code = request->phone_verify_code();
        // std::string code_id = request->phone_verify_code_id();
        // // 2. 对验证码进行验证
        // auto vcode = _redis_codes->code(code_id);
        // if (vcode != code)
        // {
        //     LOG_ERROR("{} - 验证码错误 - {}-{}！", request->request_id(), code_id, code);
        //     return err_response(request->request_id(), "验证码错误!");
        // }
        // 3. 从数据库通过用户 ID 进行用户信息查询，判断用户是否存在
        auto user = _mysql_user->select_by_id(uid);
        if (!user)
        {
            LOG_ERROR("{} - 未找到用户信息 - {}！", request->request_id(), uid);
            return err_response(request->request_id(), "未找到用户信息!");
        }
        // 4. 将新的密码更新到数据库中
        user->password(new_pwd);
        bool ret = _mysql_user->update(user);
        if (!ret)
        {
            LOG_ERROR("{} - 更新数据库用密码失败 ：{}！", request->request_id(), new_pwd);
            return err_response(request->request_id(), "更新数据库用户密码失败!");
        }
        // 5. 组织响应，返回更新成功与否
        response->set_request_id(request->request_id());
        response->set_success(true);
    }

    ~UserServiceImpl()
    {
    }

private:
    ESUser::ptr _es_user;        // es用户元信息操作对象
    UserTable::ptr _mysql_user;  // mysql用户表操作对象
    Session::ptr _redis_session; // 会话管理操作对象
    Status::ptr _redis_status;   // 用户管理操作对象
    Codes::ptr _redis_codes;     // 验证码管理对象

    // 这边是rpc调用客户端相关对象
    std::string _file_service_name;   // 文件服务名称---用于进rpc调用文件相关服务
    ServiceManager::ptr _sm_channels; // 服务信道管理对象
    DMSClient::ptr _dms_client;
};

class UserServer
{
public:
    using ptr = std::shared_ptr<UserServer>;

    UserServer(const Discover::ptr service_discoverer,
               const Register::ptr &reg_client,
               const std::shared_ptr<elasticlient::Client> &es_client,
               const std::shared_ptr<odb::core::database> &mysql_client,
               std::shared_ptr<sw::redis::Redis> &redis_client,
               const std::shared_ptr<brpc::Server> &server) : _service_discoverer(service_discoverer),
                                                              _register_client(reg_client),
                                                              _es_client(es_client),
                                                              _mysql_client(mysql_client),
                                                              _redis_client(redis_client),
                                                              _rpc_server(server) {}
    ~UserServer() {}
    // 搭建RPC服务器，并启动服务器
    void start()
    {
        _rpc_server->RunUntilAskedToQuit();
    }

private:
    Discover::ptr _service_discoverer;
    Register::ptr _register_client;
    std::shared_ptr<elasticlient::Client> _es_client;
    std::shared_ptr<odb::core::database> _mysql_client;
    std::shared_ptr<sw::redis::Redis> _redis_client;
    std::shared_ptr<brpc::Server> _rpc_server;
};

class UserServerBuilder
{
public:
    // 构造es客户端对象
    void make_es_object(const std::vector<std::string> host_list)
    {
        _es_client = ESClientFactory::create(host_list);
    }

    // 构造短信服务对象
    void make_dms_object(const std::string &access_key_id,
                         const std::string &access_key_secret)
    {
        _dms_client = std::make_shared<DMSClient>(access_key_id, access_key_secret);
    }

    // 构造mysql客户端对象
    void make_mysql_object(
        const std::string &user,
        const std::string &pswd,
        const std::string &host,
        const std::string &db,
        const std::string &cset,
        int port,
        int conn_pool_count)
    {
        _mysql_client = ODBFactory::create(user, pswd, host, db, cset, port, conn_pool_count);
    }

    // 构造redis客户端对象
    void make_redis_object(const std::string &host,
                           int port,
                           std::string user,
                           std::string pwd,
                           int db,
                           bool keep_alive)
    {
        _redis_client = RedisClientFactory::create(host, port, user, pwd, db, keep_alive);
    }

    // 用于构造服务发现客户端&信道管理对象
    void make_discovery_object(const std::string &reg_host,
                               const std::string &base_service_name,
                               const std::string &file_service_name)
    {
        _file_service_name = file_service_name;
        _sm_channels = std::make_shared<ServiceManager>();
        _sm_channels->addCareService(file_service_name);
        LOG_DEBUG("设置文件子服务为需添加管理的子服务：{}", file_service_name);
        auto put_cb = std::bind(&ServiceManager::addService, _sm_channels.get(), std::placeholders::_1, std::placeholders::_2);
        auto del_cb = std::bind(&ServiceManager::delService, _sm_channels.get(), std::placeholders::_1, std::placeholders::_2);
        _service_discoverer = std::make_shared<Discover>(reg_host, base_service_name, put_cb, del_cb);
    }
    // 用于构造服务注册客户端对象
    void make_registry_object(const std::string &reg_host,
                              const std::string &service_name,
                              const std::string &access_host)
    {
        _register_client = std::make_shared<Register>(reg_host);
        _register_client->registry(service_name, access_host);
    }
    void make_rpc_server(uint16_t port, int32_t timeout, uint8_t num_threads)
    {
        if (!_es_client)
        {
            LOG_ERROR("还未初始化ES搜索引擎模块！");
            abort();
        }
        if (!_mysql_client)
        {
            LOG_ERROR("还未初始化Mysql数据库模块！");
            abort();
        }
        if (!_redis_client)
        {
            LOG_ERROR("还未初始化Redis数据库模块！");
            abort();
        }
        if (!_sm_channels)
        {
            LOG_ERROR("还未初始化信道管理模块！");
            abort();
        }
        if (!_dms_client)
        {
            LOG_ERROR("还未初始化短信平台模块！");
            abort();
        }
        _rpc_server = std::make_shared<brpc::Server>();

        UserServiceImpl *user_service = new UserServiceImpl(_dms_client, _es_client,
                                                            _mysql_client, _redis_client, _sm_channels, _file_service_name);
        int ret = _rpc_server->AddService(user_service,
                                          brpc::ServiceOwnership::SERVER_OWNS_SERVICE);
        if (ret == -1)
        {
            LOG_ERROR("添加Rpc服务失败！");
            abort();
        }
        brpc::ServerOptions options;
        options.idle_timeout_sec = timeout;
        options.num_threads = num_threads;
        ret = _rpc_server->Start(port, &options);
        if (ret == -1)
        {
            LOG_ERROR("服务启动失败！");
            abort();
        }
    }
    // 构造RPC服务器对象
    UserServer::ptr build()
    {
        if (!_service_discoverer)
        {
            LOG_ERROR("还未初始化服务发现模块！");
            abort();
        }
        if (!_register_client)
        {
            LOG_ERROR("还未初始化服务注册模块！");
            abort();
        }
        if (!_rpc_server)
        {
            LOG_ERROR("还未初始化RPC服务器模块！");
            abort();
        }
        UserServer::ptr server = std::make_shared<UserServer>(
            _service_discoverer, _register_client,
            _es_client, _mysql_client, _redis_client, _rpc_server);
        return server;
    }

private:
    Register::ptr _register_client;

    std::shared_ptr<elasticlient::Client> _es_client;
    std::shared_ptr<odb::core::database> _mysql_client;
    std::shared_ptr<sw::redis::Redis> _redis_client;

    std::string _file_service_name;
    ServiceManager::ptr _sm_channels;
    Discover::ptr _service_discoverer;

    std::shared_ptr<DMSClient> _dms_client;

    std::shared_ptr<brpc::Server> _rpc_server;
};